import telnetlib
import time

from common.log.log import log


class TelnetClient:
    """
    telnet远程主机功能模块
    """

    def __init__(self, ):
        self.tn = telnetlib.Telnet()

    # 此函数实现telnet登录主机
    def login_host(self, host_ip, username, password):
        try:
            # self.tn = telnetlib.Telnet(host_ip,port=23)
            self.tn.open(host_ip, port=23)
        except:
            log.warning('%s网络连接失败' % host_ip)
            return False
        # 等待login出现后输入用户名，最多等待10秒
        self.tn.read_until(b'login: ', timeout=10)
        self.tn.write(username.encode('ascii') + b'\n')
        # 等待Password出现后输入密码，最多等待10秒
        self.tn.read_until(b'Password: ', timeout=10)
        self.tn.write(password.encode('ascii') + b'\n')
        # 延时两秒再收取返回结果，给服务端足够响应时间
        time.sleep(2)
        # 获取登录结果
        # read_very_eager()获取到的是的是上次获取之后本次获取之前的所有输出
        command_result = self.tn.read_very_eager().decode('ascii')
        if 'Login incorrect' not in command_result:
            log.warning('%s登录成功' % host_ip)
            return True
        else:
            log.warning('%s登录失败，用户名或密码错误' % host_ip)
            return False

    # 此函数实现执行传过来的命令，并输出其执行结果
    def execute_some_command(self, command):
        # 执行命令
        self.tn.write(command.encode('ascii') + b'\n')
        time.sleep(2)
        # 获取命令结果
        command_result = self.tn.read_very_eager().decode('ascii')
        log.warning('命令执行结果：\n%s' % command_result)

    # 退出telnet
    def logout_host(self):
        self.tn.write(b"exit\n")


def readConf(ip=''):
    """
    后面补充
    :param ip:
    :return:
    """
    pass


def del_files(command3=' *'):
    host_ip = readConf("ip")
    username = 'root'
    password = 'r'
    command1 = 'cd /root/test/'
    command3 = f'rm -rf {command3}'
    telnet_client = TelnetClient()
    # 如果登录结果返加True，则执行命令，然后退出
    if telnet_client.login_host(host_ip, username, password):
        telnet_client.execute_some_command(command1)
        time.sleep(1)
        telnet_client.execute_some_command(command3)
        time.sleep(1)
        telnet_client.logout_host()


if __name__ == '__main__':
    del_files("VarNormalTest")
